package com.atguigu.tingshu.user.service.impl;

import cn.hutool.core.date.DateUtil;
import cn.hutool.core.util.IdUtil;
import com.atguigu.tingshu.common.constant.RedisConstant;
import com.atguigu.tingshu.common.constant.SystemConstant;
import com.atguigu.tingshu.common.rabbit.constant.MqConst;
import com.atguigu.tingshu.common.rabbit.service.RabbitService;
import com.atguigu.tingshu.common.util.MongoUtil;
import com.atguigu.tingshu.model.user.UserListenProcess;
import com.atguigu.tingshu.user.service.UserListenProcessService;
import com.atguigu.tingshu.vo.album.TrackStatMqVo;
import com.atguigu.tingshu.vo.user.UserListenProcessVo;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Sort;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;

import java.math.BigDecimal;
import java.math.RoundingMode;
import java.util.Date;
import java.util.Map;
import java.util.concurrent.TimeUnit;

@Service
@SuppressWarnings({"all"})
public class UserListenProcessServiceImpl implements UserListenProcessService {

    @Autowired
    private MongoTemplate mongoTemplate;

    @Autowired
    private RedisTemplate redisTemplate;

    @Autowired
    private RabbitService rabbitService;

    /**
     * 查询用户对于指定声音播放进度
     *
     * @param userId
     * @param trackId
     * @return
     */
    @Override
    public BigDecimal getTrackBreakSecond(Long userId, Long trackId) {
        //1.确定用户播放进度集合名称
        String collectionName = this.getCollectionName(userId);

        //2.构建查询条件
        Query query = new Query();
        query.addCriteria(Criteria.where("trackId").is(trackId));

        //3.执行查询MongoDB
        UserListenProcess userListenProcess = mongoTemplate.findOne(query, UserListenProcess.class, collectionName);
        if (userListenProcess != null) {
            return userListenProcess.getBreakSecond();
        }
        return BigDecimal.valueOf(0);
    }

    /**
     * 更新用户对于指定声音播放进度
     *
     * @param userListenProcessVo
     * @param userId
     */
    @Override
    public void updateListenProcess(UserListenProcessVo userListenProcessVo, Long userId) {
        //1.确定用户播放进度集合名称
        String collectionName = this.getCollectionName(userId);

        //2.构建查询条件：声音ID。
        Query query = new Query(Criteria.where("trackId").is(userListenProcessVo.getTrackId()));
        UserListenProcess userListenProcess = mongoTemplate.findOne(query, UserListenProcess.class, collectionName);

        //3.如果存在播放进度，更新播放进度
        BigDecimal breakSeconds = userListenProcessVo.getBreakSecond().setScale(0, RoundingMode.DOWN);

        if (userListenProcess != null) {
            userListenProcess.setBreakSecond(breakSeconds);
            userListenProcess.setUpdateTime(new Date());
        } else {
            //4.如果不存在播放进度，则新增播放进度
            userListenProcess = new UserListenProcess();
            userListenProcess.setAlbumId(userListenProcessVo.getAlbumId());
            userListenProcess.setTrackId(userListenProcessVo.getTrackId());
            userListenProcess.setBreakSecond(breakSeconds);
            userListenProcess.setCreateTime(new Date());
            userListenProcess.setUpdateTime(new Date());
        }
        mongoTemplate.save(userListenProcess, collectionName);


        //TODO 5. 基于RabbitMQ更新MySQL/ES播放统计数值  MQ消息：专辑ID，声音ID，统计类型、数值
        //背景：更新播放进度业务方法，被前端每隔10S调用一次 业务要求：当天内用户对某个专辑下某个声音播放量最多加1防止刷播放量

        //5.1 确保生产消息幂等性 采用
        // 5.1.1 set nx 命令特点 将用户ID_专辑ID_声音ID作为key
        String key = RedisConstant.USER_TRACK_REPEAT_STAT_PREFIX + userId + "_" + userListenProcessVo.getAlbumId() + "_" + userListenProcessVo.getTrackId();

        //5.1.2 尝试存入Redis中，存入成功，可以发送更新统计数值MQ消息
        long ttl = DateUtil.endOfDay(new Date()).getTime() - System.currentTimeMillis();
        Boolean flag = redisTemplate.opsForValue().setIfAbsent(key, "1", ttl, TimeUnit.MILLISECONDS);
        if (flag) {
            //5.2 创建更新统计数值MQ消息对象 注意：MQ消息如果是对象，必须实现序列化接口，生成序列化ID
            TrackStatMqVo mqVo = new TrackStatMqVo();
            //业务消息不足以为消费者提供幂等处理，固设置消息唯一标识用于消费者幂等性处理
            mqVo.setBusinessNo(IdUtil.randomUUID());
            mqVo.setAlbumId(userListenProcessVo.getAlbumId());
            mqVo.setTrackId(userListenProcessVo.getTrackId());
            mqVo.setStatType(SystemConstant.TRACK_STAT_PLAY);
            mqVo.setCount(1);
            //5.3 发送MQ消息到RabbitMQ
            rabbitService.sendMessage(MqConst.EXCHANGE_TRACK, MqConst.ROUTING_TRACK_STAT_UPDATE, mqVo);
        }
        //5.1.3 反之忽略即可
    }

    /**
     * 获取用户最近播放声音
     *
     * @param userId
     * @return
     */
    @Override
    public Map<String, Long> getLatelyTrack(Long userId) {
        //1.确定用户播放进度集合名称
        String collectionName = this.getCollectionName(userId);
        //2.构建查询条件-按照更新时间排序
        Query query = new Query();
        query.with(Sort.by(Sort.Direction.DESC, "updateTime"));
        query.limit(1);
        //3.查询MongoDB获取一条记录
        UserListenProcess listenProcess = mongoTemplate.findOne(query, UserListenProcess.class, collectionName);
        if (listenProcess != null) {
            return Map.of("trackId", listenProcess.getTrackId(), "albumId", listenProcess.getAlbumId());
        }
        return null;
    }


    /**
     * 获取用户播放进度集合名称
     *
     * @param userId
     * @return
     */
    public String getCollectionName(Long userId) {
        return MongoUtil.getCollectionName(MongoUtil.MongoCollectionEnum.USER_LISTEN_PROCESS, userId);
    }
}
